import pandas as pd
import matplotlib.pyplot as plt
import matplotlib

# 确保中文字符显示正确
matplotlib.rcParams['font.sans-serif'] = ['SimHei']  # 使用SimHei字体以支持中文字符
matplotlib.rcParams['axes.unicode_minus'] = False  # 确保负号显示正确

# 加载Excel文件
file_path_res2 = "res2_亩数.xlsx"
file_path_res3 = "res3_亩数.xlsx"
df_res2 = pd.read_excel(file_path_res2)
df_res3 = pd.read_excel(file_path_res3)

# 假设数据的第二列是总亩数
total_acres_res2 = df_res2.iloc[:, 1]
total_acres_res3 = df_res3.iloc[:, 1]

# 创建图表，设置清晰度
fig, ax = plt.subplots(figsize=(14, 7), dpi=200)  # 设置dpi为150，增加图像清晰度

# X轴值：作物编号
x_values_res = list(range(1, len(total_acres_res2) + 1))

# 绘制两个数据集的折线图
ax.plot(x_values_res, total_acres_res2, label='问题二', marker='o', color='blue')
ax.plot(x_values_res, total_acres_res3, label='问题三', marker='x', color='green')

# 设置标签和标题
ax.set_xlabel('作物编号')
ax.set_ylabel('总亩数')
ax.set_title('两个数据集的作物总亩数比较')
ax.legend()

# 显示所有作物编号在x轴上
plt.xticks(x_values_res)

plt.show()
